PINN for 1D Heat Equation (PyTorch)